import pandas as pd
import torch

import warnings
warnings.filterwarnings("ignore", category=Warning)

graph_name = "Cora"
dataset_path = f"../data/{graph_name}_data.pt"
data = torch.load(dataset_path)[0]
print("Shape of y:", data['y'].shape)
